import random
import re


from flask import request, abort, current_app, make_response, jsonify

from info import redis_store, constants
from info.libs.yuntongxun.sms import CCP
from info.utils.captcha.captcha import captcha
from info.utils.response_code import RET
from . import passport_blu


@passport_blu.route('/sms_code', methods = ['POST'])
def send_sms_code():
    # 1 获取参数：手机号，团片验证码内容，图片验证码编号
    params_dict = request.json
    mobile = params_dict.get("mobile")
    image_code = params_dict.get("image_code")
    image_code_id = params_dict.get("image_code_id")
    # 2 校验参数/参数是否符合规则，是否有值
    # 验证是否有值
    if not ([mobile, image_code, image_code_id]):
        return jsonify(errno=RET.PARAMERR, errmsg="参数有误")
    # 验证手机号
    if not re.match("^[1]([3-9])[0-9]{9}$", mobile):
        return jsonify(errno=RET.PARAMERR, errmsg="手机号输入不正确")
    # 3 先从redis中取出真正的验证码内容
    try:
        real_image_code = redis_store.get("ImageCodeId" + image_code_id)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据查询失败")

    if not real_image_code:
        return jsonify(errno=RET.NODATA, errmsg="图片验证码已过期")
    # 4 与用户输入的验证码内容进行对比，如果不一致，返回验证码输入错误
    if real_image_code != image_code_id:
        return jsonify(errno=RET.DATAERR, errmsg="验证码输入有误")
    # 5 如果一致，生成验证码的内容（随机数据）
    sms_code_str = "%06d" % random.randint(0, 999999)
    current_app.logger.debug('短信验证码是:%s' % sms_code_str)
    # 6 发送短信验证码
    result = CCP().send_template_sms(mobile, [sms_code_str, constants.SMS_CODE_REDIS_EXPIRES/5],"2")

    if result != "0":
        return jsonify(errno=RET.THIRDERR, errmsg="发送短信失败")
    # 保存验证码到redis
    try:
        redis_store.set("SMS_" + mobile, sms_code_str,constants.SMS_CODE_REDIS_EXPIRES)
    except Exception as e:
        current_app.logger.error(e)
        return jsonify(errno=RET.DBERR, errmsg="数据保存失败")
    # 7 告知发送结果
    return jsonify(errno=RET.OK, errmsg="发送短信成功")


@passport_blu.route("/image_code")
def get_image_code():
    # 1 获取参数
    image_code_id = request.args.get("imageCodeId", None)

    # 2 判断参数是否有值
    if not image_code_id:
        return abort('403')
    # 3 生成图片验证码
    name, text, image = captcha.generate_captcha()

    # 4 保存图片验证码内容到redis中
    try:
        redis_store.set("ImageCodeId" + image_code_id, text, constants.IMAGE_CODE_REDIS_EXPIRES)
    except Exception as e:
        current_app.logger.error(e)
        abort("500")

    # 5 返回图片验证码
    response = make_response(image)
    response.headers['Content-Type'] = "image/jpg"
    return response
